"""
公共方法：
login个贷后台，获取cookies
"""
import requests
import time

# 请求地址
posturl = "https://ssopre.cnsuning.com/ids/login"
# 请求参数
data = {'username': '11064387', 'password': '1234567890', 'verifyCode': '', 'loginTheme': 'plsadm',
        'service': 'http://plsadmpre.cnsuning.com/plsadm/auth?targetUrl=http%3A%2F%2Fplsadmpre.cnsuning.com%2Fplsadm'
                   '%2Findex.do'}


# 获取 cookie 方法
def gl_cookie():
    # 调用requests方法请求登录
    req = requests.post(posturl, data)
    # 获取请求历史记录
    history = str(req.history)
    # 判断请求是否重定向
    while u'Response [302]' not in history:
        print(u"**********服务器异常，退出测试**********")
        break
    else:
        '''构造新cookie，提供给别的接口使用'''
        # 1.获取302重定向中的cookie  //req.request.headers 请求头中的cookie
        req_cookies = req.request.headers["Cookie"]
        # 2.获取200中的 cookie   //req.headers 响应头中的cookie
        res_session = req.headers['Set-Cookie']
        # 3.拼接cookie
        access_cookie = (res_session + '; ' + req_cookies)

        # 预置url
        rxd_url = 'http://plsadmpre.cnsuning.com/plsadm/index.do'
        # 判断登录成功与否
        while req.url != rxd_url:
            print(u"**********登录失败，退出测试**********")
        else:
            print(u"**********登录成功，等待测试**********\n")
        time.sleep(1)
        # print(access_cookie)
        return access_cookie


if __name__ == '__main__':
    gl_cookie()
